#import matplotlib.pyplot as plt

class population:
    
    def __init__(self, initialpop, birthrate,deathrate):
        self.initialpop = initialpop 
        self.birthrate = birthrate
        self.deathrate = deathrate
        self.poplist=[]

    def modelsetup(self,initialtime,finaltime,timestep): # Einstellen des Setup
        self.initialtime = initialtime
        self.finaltime = finaltime
        self.timestep = timestep
        
        

    def modelrun(self):             # hier wird das Modell laufen gelassen
        firstrun=1
        x=0
        self.durchlaeufe=((self.finaltime-self.initialtime)/self.timestep)
        
        while x <self.durchlaeufe: # Wie oft laeuft es durch?
            if firstrun==1:         # wenn das der erste Run ist, wird population so gesetzt
                population=self.initialpop
                firstrun=0
                
            else: #                 ansonsten so
                geburten=population*(self.birthrate*self.timestep)
                todesfaelle=(population*(self.deathrate*self.timestep))
                population=population+geburten-todesfaelle
                
            self.poplist.append(population)   #hier werden die Ergebnisse in eine Liste gespeichert
            x=x+1
            
    def getstock(self,anfrage):
            stock=self.poplist[int(anfrage/self.timestep)]
            print "The population in the year",anfrage,"was",stock

 #   def draw(self):
  #      for y in range(int(len(self.poplist)*self.timestep)):
  #          plt.plot([y],[self.poplist[int(y/self.timestep)]],':rs')          
  #      plt.ylabel('Population')
   #     plt.xlabel('Year')
    #    plt.show()

